from pyecharts.charts import Bar
from pyecharts import options as opts

from sc.ZDao import getConnect


def initData():
    # 连接数据库
    # 操作数据库,将数据存储进去
    db = getConnect();
    # 使用cursor()方法获取操作游标,即操作数据库的一个对象
    sql = ""
    cur = db.cursor()  # 游标 操作数据表
    #显示最新的五个城市的薪水情况
    sql ="select job_name,avg(job_sal),job_loc from jobs group by job_loc order by jid desc limit 8";
    print(sql)

    sal = [];
    ci = [];
    try:
        cur.execute(sql)
        res = cur.fetchall()  # 查到的所有的记录 是一个集合
        for r in res:
            sal.append(r[1]);
            ci.append(r[2]);
        print(sal)
        print(ci)

    except:
        db.rollback()

    bar = Bar()
    bar.add_xaxis(ci)
    bar.add_yaxis("月薪水/元", sal)
    bar.set_global_opts(title_opts=opts.TitleOpts(title="各城市-平均薪水图", subtitle="单位/元"))

    bar.render("baravg.html")
initData();